# 模型定义
model = models.resnet34
# model = models.resnet50